Update app.py

app.py

@@ -9,13 +9,14 @@ from collections import defaultdict
 from typing import List, Dict, Tuple
 
 # Configuration
-PERSISTENT_DIR = "/data/hf_cache"
-os.makedirs(os.path.join(PERSISTENT_DIR, "reports"), exist_ok=True)
+WORKING_DIR = os.getcwd()
+REPORT_DIR = os.path.join(WORKING_DIR, "reports")
+os.makedirs(REPORT_DIR, exist_ok=True)
 
 class PatientHistoryAnalyzer:
     def __init__(self):
-        self.max_token_length = 2000  # Conservative limit
-        self.max_text_length = 500    # Characters per field
+        self.max_token_length = 2000
+        self.max_text_length = 500
 
     def clean_text(self, text: str) -> str:
         """Clean and normalize text fields"""
@@ -36,7 +37,7 @@ class PatientHistoryAnalyzer:
                 'diagnoses': defaultdict(list),
                 'tests': defaultdict(list),
                 'doctors': set(),
-                'all_entries': []  # For full history analysis
+                'all_entries': []
             }
             
             for _, row in df.iterrows():
@@ -53,7 +54,6 @@ class PatientHistoryAnalyzer:
                 data['doctors'].add(entry['doctor'])
                 data['all_entries'].append(entry)
                 
-                # Categorize entries
                 form_lower = entry['form'].lower()
                 if 'medication' in form_lower or 'drug' in form_lower:
                     data['medications'][entry['item']].append(entry)
@@ -71,37 +71,34 @@ class PatientHistoryAnalyzer:
         """Generate analysis prompts that respect token limits"""
         prompts = []
         
-        # 1. Current Status Prompt (most recent data)
+        # Current Status Prompt
         current_prompt = self._create_current_status_prompt(patient_data)
         prompts.append({
             'type': 'current_status',
-            'content': current_prompt,
-            'token_estimate': len(current_prompt.split())  # Rough estimate
+            'content': current_prompt
         })
         
-        # 2. Historical Analysis Prompt (if needed)
+        # Historical Analysis Prompt
         if len(patient_data['all_entries']) > 10:
             history_prompt = self._create_historical_prompt(patient_data)
             prompts.append({
                 'type': 'historical',
-                'content': history_prompt,
-                'token_estimate': len(history_prompt.split())
+                'content': history_prompt
             })
         
-        # 3. Medication-Specific Prompt (if complex medication history)
+        # Medication-Specific Prompt
         if len(patient_data['medications']) > 3:
             meds_prompt = self._create_medication_prompt(patient_data)
             prompts.append({
                 'type': 'medications',
-                'content': meds_prompt,
-                'token_estimate': len(meds_prompt.split())
+                'content': meds_prompt
             })
         
         return prompts
 
     def _create_current_status_prompt(self, data: Dict) -> str:
         """Create prompt for current patient status"""
-        recent_entries = data['timeline'][-10:]  # Last 10 entries
+        recent_entries = data['timeline'][-10:]
         
         prompt_lines = [
             "**Comprehensive Patient Status Analysis**",
@@ -154,7 +151,7 @@ class PatientHistoryAnalyzer:
             "",
             "**Historical Timeline (condensed):**",
             *[f"- {entry['date'][:7]}: {entry['form']} - {entry['response']}" 
-              for entry in data['all_entries'][:-10]],  # All except recent 10
+              for entry in data['all_entries'][:-10]],
             "",
             "**Required Output Format:**",
             "### Historical Patterns",
@@ -196,11 +193,9 @@ class PatientHistoryAnalyzer:
             ""
         ]
         
-        # Add each analysis section
         for result in analysis_results:
             report.extend(["", "---", "", result])
         
-        # Add summary section
         report.extend([
             "",
             "## Overall Clinical Summary",
@@ -215,9 +210,11 @@ class PatientHistoryAnalyzer:
         
         full_report = "\n".join(report)
         
-        # Save to file
+        # Save to file in working directory
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        report_path = os.path.join(PERSISTENT_DIR, "reports", f"patient_report_{timestamp}.md")
+        report_filename = f"patient_report_{timestamp}.md"
+        report_path = os.path.join(REPORT_DIR, report_filename)
+        
         with open(report_path, 'w') as f:
             f.write(full_report)
         
@@ -226,26 +223,21 @@ class PatientHistoryAnalyzer:
     def analyze(self, file_path: str) -> Tuple[str, str]:
         """Main analysis workflow"""
         try:
-            # Process data
             patient_data = self.process_excel(file_path)
-            
-            # Generate prompts (simulating LLM analysis)
             prompts = self.generate_analysis_prompt(patient_data)
             
-            # Simulate LLM responses (in a real system, you'd call your LLM here)
+            # Simulate LLM responses
             simulated_responses = [
-                "### Summary of Current Status\nPatient shows improvement in blood pressure control but new concerns about medication side effects...",
-                "### Historical Patterns\nChronic back pain has been a consistent issue across 5 providers over 3 years...",
-                "### Medication Summary\nCurrent regimen includes 4 medications with one potential interaction between..."
+                "### Summary of Current Status\nPatient shows improvement in blood pressure control...",
+                "### Historical Patterns\nChronic back pain has been a consistent issue...",
+                "### Medication Summary\nCurrent regimen includes 4 medications..."
             ]
             
-            # Generate final report
             return self.generate_report(simulated_responses)
             
         except Exception as e:
             return f"Error during analysis: {str(e)}", ""
 
-# Gradio Interface
 def create_interface():
     analyzer = PatientHistoryAnalyzer()
     
@@ -261,10 +253,6 @@ def create_interface():
                             file_types=[".xlsx"],
                             type="filepath"
                         )
-                        additional_instructions = gr.Textbox(
-                            label="Special Instructions (Optional)",
-                            placeholder="E.g. 'Focus on pain management history'"
-                        )
                         analyze_btn = gr.Button("Analyze Full History", variant="primary")
                     
                     with gr.Column(scale=2):
@@ -281,13 +269,13 @@ def create_interface():
                 gr.Markdown("""
                 ## How to Use This Tool
                 
-                1. **Upload** your patient's Excel file containing all medical encounters
-                2. **Click Analyze** to process the complete history
+                1. **Upload** your patient's Excel file
+                2. **Click Analyze** to process the history
                 3. **Review** the comprehensive analysis
                 4. **Download** the full report
                 
                 ### File Requirements
-                Excel file must contain these columns:
+                Excel file must contain:
                 - Booking Number
                 - Form Name
                 - Form Item  
@@ -295,13 +283,6 @@ def create_interface():
                 - Interview Date
                 - Interviewer
                 - Description
-                
-                ### Analysis Includes
-                - Current health status
-                - Medication history
-                - Diagnostic consistency
-                - Treatment patterns
-                - Clinical recommendations
                 """)
         
         analyze_btn.click(
@@ -319,7 +300,8 @@ if __name__ == "__main__":
         demo.launch(
             server_name="0.0.0.0",
             server_port=7860,
-            show_error=True
+            show_error=True,
+            allowed_paths=[WORKING_DIR, REPORT_DIR]
         )
     except Exception as e:
         print(f"Error launching application: {str(e)}")