multi user update

Author: ANGELOANTU7

Backend/LatestSorter.py

@@ -56,19 +56,20 @@ def process_files(user:str):
     )
     print("reseting")
     # Function to read and process a file
-    def process_file(file_name):
+    def process_file(file_name,user1):
 
 
 
         # Read file from S3
-        
-        response = s3.get_object(Bucket='learnmateai', Key=user+'pyqs_txt/' + file_name)
+        print(user1)
+        response = s3.get_object(Bucket='learnmateai', Key=user1+'pyqs_txt/' + file_name)
         file_content = response['Body'].read().decode('utf-16-le')
 
         # Split file content into batches (adjust batch size as needed)
         batch_size = 30000
         batches = [file_content[i:i+batch_size] for i in range(0, len(file_content), batch_size)]
-        response2 = s3.get_object(Bucket='learnmateai', Key= user+"syllabus_pdf/syllabus.txt")
+        print(user1+"syllabus_pdf/syllabus.txt")
+        response2 = s3.get_object(Bucket='learnmateai', Key= user1+"syllabus_pdf/syllabus.txt")
         topics = response2['Body'].read().decode('utf-8')
         # Process batches
         Sorted_PYQ_Mod=[[]for _ in range(5)]
@@ -96,7 +97,7 @@ def process_file(file_name):
             #print(text_batches)
 
             bucket_name = 'learnmateai'
-            folder_name = 'Sorted_PYQS/'
+            folder_name = user1+'Sorted_PYQS/'
 
 
             i=0
@@ -117,7 +118,7 @@ def process_file(file_name):
                     s3.put_object(Bucket=bucket_name, Key=folder_name+"Module"+str(i+1)+".txt", Body=updated_content.encode('utf-8'))
 
                     # Print uploaded file information
-                    print(f"File  uploaded to '{bucket_name}/{folder_name}'")
+                    print(f"File  uploaded to '{user1}{bucket_name}/{folder_name}'")
                     i=i+1
 
 
@@ -142,7 +143,7 @@ def process_file(file_name):
                         s3.put_object(Bucket=bucket_name, Key=folder_name+"Module"+str(i+1)+".txt", Body=updated_content.encode('utf-8'))
 
                         # Print uploaded file information
-                        print(f"File  uploaded to '{bucket_name}/{folder_name}'")
+                        print(f"File  uploaded to '{user1}{bucket_name}/{folder_name}'")
                         i=i+1
                 else:
                     print("An error occurred:", e)
@@ -151,11 +152,13 @@ def process_file(file_name):
 
 
     # Get the list of files in the "notes_txt" folder
-    response = s3.list_objects_v2(Bucket='learnmateai', Prefix='pyqs_txt/')
+    response = s3.list_objects_v2(Bucket='learnmateai', Prefix=user+'pyqs_txt/')
 
     # Process each file
     for file in response['Contents']:
+        print(file)
         file_name = file['Key'].split('/')[-1]
-        process_file(file_name)
+        print(file_name)
+        process_file(file_name,user)
 
     return {"message": "PYQS SORTED"}

Backend/Notes_gen.py

@@ -14,9 +14,10 @@
 openai.api_key = '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'
 
 @app.get("/note_gen")
-async def summarize_s3_files():
+async def summarize_s3_files(user:str):
+    user=user+"/"
     bucket_name= "learnmateai"
-    folder_name= "Analysed_Notes"
+    folder_name= user+"Analysed_Notes"
     try:
         response = s3.list_objects_v2(Bucket=bucket_name, Prefix=folder_name)
         for file in response['Contents']:
@@ -25,7 +26,7 @@ async def summarize_s3_files():
             print(file_name)
             summary = await summarize_file(bucket_name, file_key,file_name)
             print(summary)
-            save_summary(file_name, summary)
+            save_summary(file_name, summary,user)
         return {'message': 'Created Notes and saved successfully.'}
     except Exception as e:
         return {'error': str(e)}
@@ -50,10 +51,10 @@ async def summarize_file(bucket_name: str, file_key: str, file_name:str):
     except Exception as e:
         raise e
 
-def save_summary(file_name: str, summary: str):
+def save_summary(file_name: str, summary: str,user):
     try:
         file_name=file_name.split(".txt")[0]
-        save_key = f'Notes_Topicwise/{file_name}.txt'
+        save_key = f'{user}Notes_Topicwise/{file_name}.txt'
         s3.put_object(Body=summary, Bucket=s3_bucket_name, Key=save_key)
     except Exception as e:
         raise e

Backend/Sections_topics_json.py

@@ -30,10 +30,11 @@ def processor( data):
     return summary
 
 @app.get("/card-json")
-async def makejson():
+async def makejson(user:str):
+    user=user+"/"
     text=""
     bucket_name= "learnmateai"
-    folder_name= "Analysed_Notes"
+    folder_name= user+"Analysed_Notes"
     try:
         response = s3.list_objects_v2(Bucket=bucket_name, Prefix=folder_name)
         for file in response['Contents']:
@@ -46,7 +47,7 @@ async def makejson():
         print(text)
 
         json_text=processor(text)    
-        save_plan(json_text)
+        save_plan(json_text,user)
             # Parse the file content as JSON
         json_content = json.loads(json_text)
 
@@ -58,9 +59,9 @@ async def makejson():
 
 
 
-def save_plan( summary: str):
+def save_plan( summary: str,user):
     try:
-        save_key = f'Cardjson/cards.txt'
+        save_key = f'{user}Cardjson/cards.txt'
         s3.put_object(Body=summary, Bucket=s3_bucket_name, Key=save_key)
     except Exception as e:
         raise e