Update app.py

app.py

@@ -12,7 +12,7 @@ model = None
 tokenizer = None
 generator = None
 csv_name = "disease_database_mini.csv"
-globals df 
+global df 
 df = pd.read_csv(csv_name)
 
 
@@ -115,7 +115,10 @@ def csv_prompter(question,csv_name):
     return answer_llm
 
 
-
+def filter_records(records):
+    df = records[1:]
+    return df
+    
 
 with gr.Blocks() as demo:
     gr.Markdown("## Autonomous ChatDoctor (openai version), based on disease database knowledge")
@@ -159,6 +162,17 @@ with gr.Blocks() as demo:
     clear.click(lambda: None, None, chatbot, queue=False).then(restart, chatbot, chatbot)
     Initialization.click(lambda: None, None, chatbot, queue=False).then(restart, chatbot, chatbot)
 
+    
+    gr.Interface(
+    filter_records,
+    [
+        gr.Dataframe(
+            df
+        ),
+    ],
+    "dataframe",
+    description="Disease Database",)
+    
 if __name__ == "__main__":
     demo.launch()