Update app.py

app.py

@@ -8,8 +8,7 @@ def set_api_key(api_key):
     if api_key:
         openai.api_key = api_key
         return "API Key Set"
-    else:
-        return "API Key not set"
+    return "API Key not set"
 
 # PDF text extraction function
 def extract_text_from_pdf(pdf_path):
@@ -24,19 +23,23 @@ def extract_text_from_pdf(pdf_path):
 # Review generator function
 def generate_review(api_key, uploaded_files, review_question, include_tables):
     if not api_key:
-        return "Please enter your OpenAI API key."
+        return "Please enter your OpenAI API key.", None
     if not uploaded_files:
-        return "Please upload at least one PDF file."
+        return "Please upload at least one PDF file.", None
     if not review_question:
-        return "Please enter a review question."
+        return "Please enter a review question.", None
     
-    # File type validation (redundant but kept for robustness)
+    # Ensure uploaded_files is a list of file paths
+    if not isinstance(uploaded_files, list):
+        uploaded_files = [uploaded_files]
+    
+    # Validate file types
     for file in uploaded_files:
-        if not file.lower().endswith('.pdf'):
-            return f"Invalid file type. Please upload a PDF file, but '{os.path.basename(file)}' is not a PDF."
+        if not isinstance(file, str) or not file.lower().endswith('.pdf'):
+            return f"Invalid file type detected. Please upload only PDF files. Problematic file: '{file}'", None
     
     system_prompt = """
-    You are an expert academic assistant. Create a systematic review in HTML format using <h2>, <h3>, <p>, <ul>, and <table> tags. The Systematic Review must be in great details. Structure it using these steps:
+    You are an expert academic assistant. Create a systematic review in HTML format using <h2>, <h3>, <p>, <ul>, and <table> tags. The Systematic Review must be in great detail. Structure it using these steps:
     Step 1: Identify a Research Field
     The first step in writing a systematic review paper is to identify a research field. This involves selecting a specific area of study that you are interested in and want to explore further.
     Step 2: Generate a Research Question
@@ -66,7 +69,7 @@ def generate_review(api_key, uploaded_files, review_question, include_tables):
     # Check for extraction errors
     for i, text in enumerate(texts):
         if text.startswith("Error extracting text"):
-            return text
+            return text, None
 
     table_note = " Include relevant tables to compare methodologies, results, and limitations." if include_tables else ""
     user_prompt = (
@@ -77,36 +80,35 @@ def generate_review(api_key, uploaded_files, review_question, include_tables):
 
     try:
         response = openai.ChatCompletion.create(
-            model="gpt-4",  # Updated from "gpt-4.1" to a valid model
+            model="gpt-4",
             messages=[{"role": "system", "content": system_prompt}, {"role": "user", "content": user_prompt}],
             temperature=0.7,
             top_p=1,
             max_tokens=8192
         )
         review_html = response["choices"][0]["message"]["content"]
-        download_button = gr.File(file_name="systematic_review.html", data=review_html)
+        download_button = review_html.encode('utf-8')  # For file download
 
-        return [review_html, download_button]
+        return review_html, download_button
     except Exception as e:
-        return f"Error generating systematic review: {e}"
+        return f"Error generating systematic review: {e}", None
 
 # Gradio Interface
-api_key_input = gr.Textbox(label="API Key", type="password", placeholder="Enter OpenAI API Key", lines=1)
-file_input = gr.File(label="Upload PDF Research Papers", file_count="multiple", file_types=["pdf"])
-review_question_input = gr.Textbox(label="Review Question or Topic", placeholder="Please Generate a systematic review of the following papers.")
-include_tables_input = gr.Checkbox(label="Include Comparison Tables", value=True)
-
-output_review = gr.HTML(label="Generated Systematic Review")
-output_download = gr.File(label="Download Review as .html")
-
-# Create Interface
-interface = gr.Interface(
-    fn=generate_review,
-    inputs=[api_key_input, file_input, review_question_input, include_tables_input],
-    outputs=[output_review, output_download],
-    live=True,
-    title="Systematic Review Generator for Research Papers",
-    description="Generate a polished and structured systematic review from multiple uploaded research papers."
-)
+with gr.Blocks(title="Systematic Review Generator for Research Papers") as interface:
+    gr.Markdown("# Systematic Review Generator\nUpload PDF research papers to generate a structured systematic review.")
+    
+    api_key_input = gr.Textbox(label="API Key", type="password", placeholder="Enter OpenAI API Key")
+    file_input = gr.File(label="Upload PDF Research Papers", file_count="multiple", file_types=["pdf"])
+    review_question_input = gr.Textbox(label="Review Question or Topic", placeholder="E.g., What are the impacts of climate change on agriculture?")
+    include_tables_input = gr.Checkbox(label="Include Comparison Tables", default=True)
+    
+    output_review = gr.HTML(label="Generated Systematic Review")
+    output_download = gr.File(label="Download Review as .html")
+    
+    gr.Button("Generate Review").click(
+        fn=generate_review,
+        inputs=[api_key_input, file_input, review_question_input, include_tables_input],
+        outputs=[output_review, output_download]
+    )
 
 interface.launch()