Spaces:

sonuprasad23
/

quantum-bot

Running

App Files Files Community

sonuprasad23 commited on Sep 14, 2025

Commit

9286802

1 Parent(s): da2774c

Almost Done

Browse files

Files changed (1) hide show

server.py +44 -39

server.py CHANGED Viewed

@@ -137,8 +137,8 @@ def run_automation_process(session_id):
         if workflow == 'void':
             results = bot_instance.process_void_list(patient_data)
         elif workflow == 'refund':
-            date_range = {'start_date': data.get('start_date'), 'end_date': data.get('end_date')}
-            results = bot_instance.process_refund_list(patient_data, date_range)
         is_terminated = bot_instance.termination_event.is_set()
     except Exception as e:
         print(f"[Server Log] Fatal error in automation thread: {e}"); is_crash = True
@@ -194,57 +194,62 @@ def status_page():
     APP_STATUS_HTML = """<!DOCTYPE html><html lang="en"><head><title>API Status</title><style>body{font-family:sans-serif;display:flex;justify-content:center;align-items:center;height:100vh;margin:0;background:#f0f2f5;}.status-box{text-align:center;padding:40px 60px;background:white;border-radius:12px;box-shadow:0 8px 30px rgba(0,0,0,0.1);}h1{font-size:24px;color:#333;margin-bottom:10px;} .indicator{font-size:18px;font-weight:600;padding:8px 16px;border-radius:20px;}.active{color:#28a745;background-color:#e9f7ea;}</style></head><body><div class="status-box"><h1>Hillside Automation API</h1><div class="indicator active">● Active</div></div></body></html>"""
     return Response(APP_STATUS_HTML)
-def extract_patient_name(raw_name):
-    if not isinstance(raw_name, str): return ""
-    name_only = raw_name.split('DOB')[0].strip()
-    return re.sub(r'[:\d\-\s]+$', '', name_only).strip()
-@app.route('/process_files_for_automation', methods=['POST'])
-def handle_file_processing():
     session_id = 'user_session'
     try:
-        if 'app_data' not in request.files or 'quantum_data' not in request.files:
-            return jsonify({"error": "Both files are required."}), 400
-        df_app = pd.read_excel(request.files['app_data']) if request.files['app_data'].filename.endswith('.xlsx') else pd.read_csv(request.files['app_data'])
-        df_quantum = pd.read_excel(request.files['quantum_data']) if request.files['quantum_data'].filename.endswith('.xlsx') else pd.read_csv(request.files['quantum_data'])
-        if 'Patient Name' not in df_app.columns or 'PRN' not in df_app.columns: raise ValueError("App Data must contain 'Patient Name' and 'PRN' columns.")
-        if 'Name' not in df_quantum.columns: raise ValueError("Quantum Data must contain a 'Name' column.")
-        df_app_filtered = df_app.dropna(subset=['PRN']); df_app_filtered = df_app_filtered[df_app_filtered['PRN'].astype(str).str.strip() != '']
         prn_lookup_dict = {extract_patient_name(row['Patient Name']): row['PRN'] for _, row in df_app_filtered.iterrows()}
         master_df = df_quantum.copy()
         master_df['PRN'] = master_df['Name'].apply(lambda name: prn_lookup_dict.get(name, ""))
         master_df['Status'] = ''
         session_data[session_id]['patient_data_for_report'] = master_df
         session_data[session_id]['patient_data'] = master_df.to_dict('records')
         socketio.emit('data_processed')
         print(f"[Server Log] Data prepared. Total records: {len(master_df)}")
-        return jsonify({"message": "Data processed successfully."})
     except Exception as e:
-        print(f"[Server Log] ERROR during file processing: {e}")
-        return jsonify({"error": str(e)}), 500
-@socketio.on('connect')
-def handle_connect():
-    print(f'Frontend connected.')
-    emit('email_list', {'emails': get_email_list()})
-@socketio.on('initialize_session')
-def handle_init(data):
-    session_id = 'user_session'
-    session_data[session_id] = {
-        'emails': data['emails'],
-        'filename': data['filename'],
-        'workflow': data['workflow'],
-        'start_date': data.get('start_date'),
-        'end_date': data.get('end_date')
-    }
 @socketio.on('start_login')
 def handle_login(credentials):
     global bot_instance
@@ -278,4 +283,4 @@ if __name__ == '__main__':
     print(f"  Frontend URL: {FRONTEND_ORIGIN}")
     print(f"  Port: {os.getenv('PORT', 7860)}")
     print("====================================================================")
-    socketio.run(app, host='0.0.0.0', port=int(os.getenv('PORT', 7860)))

         if workflow == 'void':
             results = bot_instance.process_void_list(patient_data)
         elif workflow == 'refund':
+            start_date = data.get('start_date'); end_date = data.get('end_date')
+            results = bot_instance.process_refund_list(patient_data, start_date, end_date)
         is_terminated = bot_instance.termination_event.is_set()
     except Exception as e:
         print(f"[Server Log] Fatal error in automation thread: {e}"); is_crash = True
     APP_STATUS_HTML = """<!DOCTYPE html><html lang="en"><head><title>API Status</title><style>body{font-family:sans-serif;display:flex;justify-content:center;align-items:center;height:100vh;margin:0;background:#f0f2f5;}.status-box{text-align:center;padding:40px 60px;background:white;border-radius:12px;box-shadow:0 8px 30px rgba(0,0,0,0.1);}h1{font-size:24px;color:#333;margin-bottom:10px;} .indicator{font-size:18px;font-weight:600;padding:8px 16px;border-radius:20px;}.active{color:#28a745;background-color:#e9f7ea;}</style></head><body><div class="status-box"><h1>Hillside Automation API</h1><div class="indicator active">● Active</div></div></body></html>"""
     return Response(APP_STATUS_HTML)
+@socketio.on('connect')
+def handle_connect():
+    print(f'Frontend connected.')
+    emit('email_list', {'emails': get_email_list()})
+@socketio.on('get_email_list')
+def handle_get_email_list():
+    emit('email_list', {'emails': get_email_list()})
+@socketio.on('initialize_session')
+def handle_init(data):
     session_id = 'user_session'
+    session_data.setdefault(session_id, {})
+    session_data[session_id].update({
+        'emails': data.get('emails', []), 'filename': data.get('filename'),
+        'workflow': data.get('workflow'), 'start_date': data.get('start_date'),
+        'end_date': data.get('end_date'),
+    })
     try:
+        app_b64 = data.get('app_data_b64'); quantum_b64 = data.get('quantum_data_b64')
+        if not app_b64 or not quantum_b64:
+            emit('error', {'message': 'Both data files are required.'}); return
+        def load_df_from_b64(b64str):
+            raw = base64.b64decode(b64str); bio = io.BytesIO(raw)
+            try: return pd.read_excel(bio)
+            except Exception: bio.seek(0); return pd.read_csv(bio)
+        df_app = load_df_from_b64(app_b64); df_quantum = load_df_from_b64(quantum_b64)
+        if 'Patient Name' not in df_app.columns or 'PRN' not in df_app.columns:
+            emit('error', {'message': "App Data must contain 'Patient Name' and 'PRN' columns."}); return
+        if 'Name' not in df_quantum.columns:
+            emit('error', {'message': "Quantum Data must contain a 'Name' column."}); return
+        def extract_patient_name(raw_name):
+            if not isinstance(raw_name, str): return ""
+            name_only = raw_name.split('DOB')[0].strip()
+            return re.sub(r'[:\d\-\s]+$', '', name_only).strip()
+        df_app_filtered = df_app.dropna(subset=['PRN'])
+        df_app_filtered = df_app_filtered[df_app_filtered['PRN'].astype(str).str.strip() != '']
         prn_lookup_dict = {extract_patient_name(row['Patient Name']): row['PRN'] for _, row in df_app_filtered.iterrows()}
         master_df = df_quantum.copy()
         master_df['PRN'] = master_df['Name'].apply(lambda name: prn_lookup_dict.get(name, ""))
         master_df['Status'] = ''
         session_data[session_id]['patient_data_for_report'] = master_df
         session_data[session_id]['patient_data'] = master_df.to_dict('records')
         socketio.emit('data_processed')
         print(f"[Server Log] Data prepared. Total records: {len(master_df)}")
     except Exception as e:
+        emit('error', {'message': f'Error preparing data: {e}'})
 @socketio.on('start_login')
 def handle_login(credentials):
     global bot_instance
     print(f"  Frontend URL: {FRONTEND_ORIGIN}")
     print(f"  Port: {os.getenv('PORT', 7860)}")
     print("====================================================================")
+    socketio.run(app, host='0.0.0.0', port=int(os.getenv('PORT', 7860)))```