Add some updates

Dockerfile

@@ -50,19 +50,47 @@ COPY . .
 # Set proper ownership of all files
 RUN chown -R appuser:appuser /app
 
-# Create a startup script
+# Create a startup script with better logging and error handling
 RUN echo '#!/bin/bash\n\
+set -e\n\
+\n\
+echo "Starting Healthcare AI Assistant..."\n\
+\n\
+# Function to check if Ollama is running\n\
+check_ollama() {\n\
+    curl -s http://localhost:11434/api/tags > /dev/null\n\
+    return $?\n\
+}\n\
+\n\
 # Start Ollama in the background\n\
+echo "Starting Ollama server..."\n\
 ollama serve &\n\
+OLLAMA_PID=$!\n\
 \n\
 # Wait for Ollama to start\n\
-sleep 10\n\
+echo "Waiting for Ollama to start..."\n\
+for i in {1..30}; do\n\
+    if check_ollama; then\n\
+        echo "Ollama server is running"\n\
+        break\n\
+    fi\n\
+    if [ $i -eq 30 ]; then\n\
+        echo "Error: Ollama server failed to start"\n\
+        exit 1\n\
+    fi\n\
+    sleep 1\n\
+done\n\
 \n\
 # Pull Mistral model\n\
+echo "Pulling Mistral model..."\n\
 ollama pull mistral\n\
 \n\
 # Start FastAPI application\n\
-cd /app && uvicorn main:app --host 0.0.0.0 --port 8000\n\
+echo "Starting FastAPI application..."\n\
+cd /app\n\
+\n\
+# Keep the container running and show logs\n\
+exec uvicorn main:app --host 0.0.0.0 --port 8000 --log-level info\n\
 ' > /app/start.sh && chmod +x /app/start.sh
 
 # Expose ports

final.py

@@ -689,20 +689,20 @@ class AIAgent:
 
 
 # Test the agent directly
-if __name__ == "__main__":
-    agent = AIAgent()
+# if __name__ == "__main__":
+#     agent = AIAgent()
     
-    # Test with English query
-    # print("\n---Testing English Query---")
-    # english_response = agent.process_user_query("I need to book an appointment with Dr. Smith tomorrow at 8 PM")
-    # print("\nEnglish response:")
-    # print(english_response["user_friendly_response"])
+#     # Test with English query
+#     # print("\n---Testing English Query---")
+#     # english_response = agent.process_user_query("I need to book an appointment with Dr. Smith tomorrow at 8 PM")
+#     # print("\nEnglish response:")
+#     # print(english_response["user_friendly_response"])
     
-    # Test with Arabic query
-    print("\n---Testing Arabic Query---")
-    # arabic_response = agent.process_user_query(" اريد الغاء الحجز مع الدكتور Smith")
-    arabic_response = agent.process_user_query("اريد حجز ميعاد غدا في الساعه الثامنه مساء مع الدكتور Smith")
-    # arabic_response = agent.process_user_query("متى يفتح المستشفى؟")
-    # arabic_response = agent.process_user_query("اريد معرفه كل الحجوزات الخاصه بي")
-    print("\nArabic response:")
-    print(arabic_response["user_friendly_response"])
+#     # Test with Arabic query
+#     print("\n---Testing Arabic Query---")
+#     # arabic_response = agent.process_user_query(" اريد الغاء الحجز مع الدكتور Smith")
+#     arabic_response = agent.process_user_query("اريد حجز ميعاد غدا في الساعه الثامنه مساء مع الدكتور Smith")
+#     # arabic_response = agent.process_user_query("متى يفتح المستشفى؟")
+#     # arabic_response = agent.process_user_query("اريد معرفه كل الحجوزات الخاصه بي")
+#     print("\nArabic response:")
+#     print(arabic_response["user_friendly_response"])