Initial

app.py

@@ -113,7 +113,7 @@ def parse_xml_response(xml_string: str) -> pd.DataFrame:
         print(f"Traceback: {traceback.format_exc()}")
         return pd.DataFrame()
 
-def get_response(prompt, file_id, assistant_id):
+def get_response(file_id, assistant_id):
     try:
         thread = client.beta.threads.create(
             messages=[
@@ -153,7 +153,7 @@ def process(file_content):
 
         message_file = client.files.create(file=open(file_name, "rb"), purpose="assistants")
 
-        response = get_response(prompts.info_prompt, message_file.id, demo.id)
+        response = get_response(message_file.id, demo.id)
         df = parse_xml_response(response)
         
         if df.empty:
@@ -216,9 +216,47 @@ def gradio_interface():
     demo.queue()
     demo.launch()
 
+def run_in_terminal():
+    print("Clinical Note Information Extractor")
+    print("This tool extracts key information from clinical notes in PDF format.")
+    print("Enter the path to your PDF file:")
+    file_path = input().strip()
+
+    if not os.path.exists(file_path):
+        print(f"Error: File not found at {file_path}")
+        return
+
+    try:
+        with open(file_path, "rb") as file:
+            file_content = file.read()
+        
+        result = process(file_content)
+        
+        if result.startswith("<p>"):
+            # Error message
+            print(result[3:-4])  # Remove <p> tags
+        else:
+            # Save the HTML output to a file
+            output_file = f"output_{time.time()}.html"
+            with open(output_file, "w", encoding="utf-8") as f:
+                f.write(result)
+            print(f"Extraction completed. Results saved to {output_file}")
+            
+            # Also print a simplified version to the console
+            df = pd.read_html(result)[0]
+            print("\nExtracted Information:")
+            for _, row in df.iterrows():
+                print(f"{row['Category']} - {row['Field']}: {row['Value']}")
+
+    except Exception as e:
+        print(f"An error occurred while processing the file: {str(e)}")
+        print(f"Traceback: {traceback.format_exc()}")
+
+
 if __name__ == "__main__":
     try:
         gradio_interface()
+        # run_in_terminal()
     except Exception as e:
         print(f"Error launching Gradio interface: {str(e)}")
         print(f"Traceback: {traceback.format_exc()}")