Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import os
-import base64
+import tempfile
 from openai import OpenAI
 
 def generate_systematic_review(api_key, pdf_files):
@@ -48,60 +48,82 @@ The final step is to make concluding statements. This involves summarizing your
 By following these steps, you can ensure that your systematic review paper is well-written, well-organized, and provides valuable insights into your research question.
 """
         
-        # Create a list to store file data for the API call
-        file_contents = []
-        
-        # Process each uploaded PDF file
+        # First, upload all files to get file IDs
+        file_ids = []
         for pdf_file in pdf_files:
-            file_name = os.path.basename(pdf_file.name)
-            
-            # Read the file as binary data
             with open(pdf_file.name, "rb") as f:
-                binary_data = f.read()
-            
-            # Encode the binary data to base64
-            base64_encoded = base64.b64encode(binary_data).decode('utf-8')
-            
-            # Create proper data URL with MIME type
-            file_data = f"data:application/pdf;base64,{base64_encoded}"
-            
-            # Add to file contents
-            file_contents.append({
-                "type": "file_attachment",
-                "file_name": file_name,
-                "data": file_data
-            })
+                file_response = client.files.create(
+                    file=f,
+                    purpose="assistants"
+                )
+                file_ids.append(file_response.id)
+        
+        # Create an assistant
+        assistant = client.beta.assistants.create(
+            name="Systematic Review Generator",
+            instructions=system_prompt,
+            model="gpt-4.1",
+            tools=[{"type": "file_search"}],
+            file_ids=file_ids
+        )
         
-        # Create messages for the API call
-        messages = [
-            {
-                "role": "system",
-                "content": system_prompt
-            },
-            {
-                "role": "user",
-                "content": [
-                    {"type": "text", "text": "Please generate the systematic review of these papers (include also important new generated tables)"},
-                    *file_contents
-                ]
-            }
-        ]
+        # Create a thread
+        thread = client.beta.threads.create()
         
-        # Make the API call to OpenAI
-        response = client.chat.completions.create(
-            model="gpt-4.1",  # Updated model name (or use gpt-4-turbo)
-            messages=messages,
-            temperature=0.7,
-            max_tokens=4000,
-            top_p=1
+        # Add a message to the thread
+        message = client.beta.threads.messages.create(
+            thread_id=thread.id,
+            role="user",
+            content="Please generate the systematic review of these papers (include also important new generated tables)"
         )
         
-        # Extract and return the review text from the response
-        if response.choices and len(response.choices) > 0:
-            result_text = response.choices[0].message.content
-            if result_text:
-                return result_text
+        # Run the assistant on the thread
+        run = client.beta.threads.runs.create(
+            thread_id=thread.id,
+            assistant_id=assistant.id
+        )
         
+        # Poll for the run completion
+        import time
+        while True:
+            run_status = client.beta.threads.runs.retrieve(
+                thread_id=thread.id,
+                run_id=run.id
+            )
+            if run_status.status == "completed":
+                break
+            elif run_status.status in ["failed", "cancelled", "expired"]:
+                return f"Run failed with status: {run_status.status}"
+            time.sleep(2)
+        
+        # Get the messages from the thread
+        messages = client.beta.threads.messages.list(thread_id=thread.id)
+        
+        # Get the latest assistant message
+        assistant_messages = [msg for msg in messages.data if msg.role == "assistant"]
+        if assistant_messages:
+            latest_message = assistant_messages[0]  # Most recent message first
+            
+            # Extract text content from the message
+            result_text = ""
+            for content_item in latest_message.content:
+                if content_item.type == "text":
+                    result_text += content_item.text.value
+            
+            # Clean up: delete assistant, thread, and files
+            client.beta.assistants.delete(assistant_id=assistant.id)
+            client.beta.threads.delete(thread_id=thread.id)
+            for file_id in file_ids:
+                client.files.delete(file_id=file_id)
+                
+            return result_text
+        
+        # Clean up resources even if we didn't get a response
+        client.beta.assistants.delete(assistant_id=assistant.id)
+        client.beta.threads.delete(thread_id=thread.id)
+        for file_id in file_ids:
+            client.files.delete(file_id=file_id)
+            
         return "Failed to generate a systematic review. Please try again."
         
     except Exception as e:
@@ -153,7 +175,7 @@ with gr.Blocks(title="Systematic Review Generator") as app:
     This application requires a valid OpenAI API key with access to the GPT-4 model. 
     Your API key is not stored and is only used to make the API call to OpenAI.
     
-    Please note that large PDF files may cause issues with the API due to size limits.
+    The review generation may take a few minutes depending on the number and size of PDF files.
     """)
 
 if __name__ == "__main__":