Update app.py

app.py

@@ -196,12 +196,14 @@ def gradio_interface(search_str, selected_names, user_query):
     models = search_models(search_str)
     
     if not models:
-        return "No models found for the given search query.", None
+        return "No models found for the given search query.", [], None
+    
+    model_names = [model_data['name'] for model_id, model_data in models.items()]
     
     selected_model_ids = [model_id for model_id, model_data in models.items() if model_data['name'] in selected_names]
     
     if not selected_model_ids:
-        return "No models selected for analysis.", None
+        return "No models selected for analysis.", model_names, None
     
     all_final_items = []
     for model_id in selected_model_ids:
@@ -215,7 +217,7 @@ def gradio_interface(search_str, selected_names, user_query):
         
         final_items = split_biomodels(antimony_file_path)
         if not final_items:
-            return "No content found in the biomodel.", None
+            return "No content found in the biomodel.", model_names, None
         
         all_final_items.extend(final_items)
     
@@ -224,9 +226,9 @@ def gradio_interface(search_str, selected_names, user_query):
     
     if db:
         response = generate_response(db, user_query, "")
-        return "Models have been processed and added to the database.", response
+        return "Models have been processed and added to the database.", model_names, response
     
-    return "Database creation failed.", None
+    return "Database creation failed.", model_names, None
 
 def main():
     gr.Interface(
@@ -238,10 +240,12 @@ def main():
         ],
         outputs=[
             gr.Textbox(label="Status"),
+            gr.CheckboxGroup(label="Select biomodels", choices=[]),
             gr.Textbox(label="Response")
         ],
         live=True
-    ).launch()
+    ).launch(share=True)
 
 if __name__ == "__main__":
     main()
+