Update app.py

app.py

@@ -157,7 +157,7 @@ Format the response clearly with headings for "Recommended Medicines", "Medicine
     except Exception as e:
         return f"Error: {str(e)}"
 
-def send_medicine_to_recommender(api_key, medicine_names, csv_file=None):
+def send_medicine_to_recommender(api_key, medicine_names, csv_file):
     """
     Takes medicine names extracted from prescription and gets recommendations
     """
@@ -193,6 +193,14 @@ with gr.Blocks(title="Medicine Assistant") as app:
         type="password"
     )
     
+    # Create a file input for CSV that can be shared between tabs
+    csv_file_input = gr.File(
+        label="Upload Medicine CSV (Optional)", 
+        file_types=[".csv"],
+        type="file"
+    )
+    gr.Markdown("If no CSV is uploaded, the app will use the default 'medicine_dataset.csv' file.")
+    
     with gr.Tabs():
         with gr.Tab("Prescription Medicine Extractor"):
             gr.Markdown("## Prescription Medicine Extractor")
@@ -217,7 +225,7 @@ with gr.Blocks(title="Medicine Assistant") as app:
             
             recommend_from_extract_btn.click(
                 fn=send_medicine_to_recommender,
-                inputs=[api_key_input, extracted_output, None],  # Pass None as csv_file to use default
+                inputs=[api_key_input, extracted_output, csv_file_input],  # Now correctly passing the csv_file_input
                 outputs=recommendation_from_extract_output
             )
             
@@ -226,7 +234,8 @@ with gr.Blocks(title="Medicine Assistant") as app:
             1. Enter your Together API key
             2. Upload a clear image of a prescription
             3. Click 'Extract Medicines' to see the identified medicines
-            4. Optionally click 'Get Recommendations for First Medicine' to find alternatives
+            4. Optionally upload a custom medicine dataset CSV
+            5. Click 'Get Recommendations for First Medicine' to find alternatives
             
             ### Note:
             - Your API key is used only for the current session
@@ -243,13 +252,6 @@ with gr.Blocks(title="Medicine Assistant") as app:
                         label="Medicine Name", 
                         placeholder="Enter a medicine name exactly as it appears in the dataset"
                     )
-                    csv_file = gr.File(
-                        label="Upload Medicine CSV (Optional)", 
-                        file_types=[".csv"],
-                        type="file"
-                    )
-                    gr.Markdown("If no CSV is uploaded, the app will use the default 'medicine_dataset.csv' file.")
-                    
                     submit_btn = gr.Button("Get Recommendations", variant="primary")
                 
                 with gr.Column():
@@ -257,7 +259,7 @@ with gr.Blocks(title="Medicine Assistant") as app:
             
             submit_btn.click(
                 recommend_medicine,
-                inputs=[api_key_input, medicine_name, csv_file],
+                inputs=[api_key_input, medicine_name, csv_file_input],  # Using the shared csv_file_input
                 outputs=recommendation_output
             )
 
@@ -265,8 +267,7 @@ with gr.Blocks(title="Medicine Assistant") as app:
             ## How to use this tool:
             1. Enter your Together API key (same key used across the application)
             2. Enter a medicine name **exactly as it appears** in the CSV file
-            3. Optionally upload a custom medicine dataset CSV file (otherwise the default medicine_dataset.csv will be used)
-            4. Click "Get Recommendations" to see alternatives
+            3. Click "Get Recommendations" to see alternatives
             
             ### CSV Format Requirements:
             The app expects a CSV with these columns: