Resolving docker error

app.py

@@ -1,9 +1,8 @@
-import gradio as gr
 from doctr.models import ocr_predictor
 from PIL import Image
 import numpy as np
 import torch
-from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_huggingface import HuggingFaceEmbeddings
 from langchain_core.documents import Document
 import os
 from groq import Groq
@@ -302,7 +301,7 @@ def query_qdrant_store(query_text, k=3):
         retrieved_docs.append({
             "source": doc.metadata['source'],
             "content": doc.page_content,
-            "score": score,
+            "score": float(score),
             "metadata": doc.metadata
         })
     return retrieved_docs
@@ -412,26 +411,38 @@ def add_documents_to_qdrant(docs):
         raise
 
 # -------------------------------
-# Flask App Setup
+# Flask App - API ONLY
 # -------------------------------
-flask_app = Flask(__name__)
+app = Flask(__name__)
 
-@flask_app.route('/api/query', methods=['POST'])
+@app.route('/', methods=['GET'])
+def home():
+    return jsonify({
+        "message": "🧠 Multimodal RAG API",
+        "endpoints": {
+            "/query": "POST - Query documents",
+            "/upload": "POST - Upload files",
+            "/health": "GET - Health check"
+        }
+    })
+
+@app.route('/query', methods=['POST'])
 def handle_query():
     data = request.get_json()
-    query = data.get('query', '')
-    k = data.get('k', 3)
     
-    if not query:
+    if not data or 'query' not in data:
         return jsonify({"error": "No query provided"}), 400
     
+    query = data.get('query', '')
+    k = data.get('k', 3)
+    
     try:
         response_data = get_rag_response(query, k)
         return jsonify(response_data)
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-@flask_app.route('/api/upload', methods=['POST'])
+@app.route('/upload', methods=['POST'])
 def handle_upload():
     if 'files' not in request.files:
         return jsonify({"error": "No files provided"}), 400
@@ -476,88 +487,15 @@ def handle_upload():
         "failed_files": failed_files
     })
 
-@flask_app.route('/api/health', methods=['GET'])
+@app.route('/health', methods=['GET'])
 def health_check():
     return jsonify({"status": "ok", "message": "API is running"})
 
-# -------------------------------
-# Gradio UI
-# -------------------------------
-def gradio_chat_response(message, history):
-    response_data = get_rag_response(message, k=3)
-    answer = response_data['answer']
-    sources = response_data['sources']
-    
-    sources_md = "\n\n---\n**📚 Sources:**\n"
-    for doc in sources:
-        sources_md += f"* {doc['source']} (Score: {doc['score']:.4f})\n"
-    
-    return answer + sources_md
-
-def gradio_upload(file_list):
-    if not file_list:
-        return "No files uploaded."
-    
-    docs_to_add = []
-    processed = 0
-    failed = 0
-    
-    for file_obj in file_list:
-        full_path = file_obj.name
-        filename = os.path.basename(full_path)
-        
-        try:
-            doc = process_single_file(full_path, filename)
-            if doc:
-                docs_to_add.append(doc)
-                processed += 1
-            else:
-                failed += 1
-        except Exception as e:
-            print(f"❌ Error: {e}")
-            failed += 1
-    
-    if docs_to_add:
-        try:
-            add_documents_to_qdrant(docs_to_add)
-        except Exception as e:
-            return f"❌ Error: {e}"
-    
-    return f"✅ Processed {processed} files. Failed: {failed}."
-
-with gr.Blocks(theme="soft") as gradio_ui:
-    gr.Markdown("# 🧠 Multimodal RAG System")
-    
-    with gr.Tabs():
-        with gr.TabItem("💬 Chat"):
-            gr.ChatInterface(
-                fn=gradio_chat_response,
-                title="Chat with Documents",
-                description="Ask questions about your documents",
-                examples=[
-                    "What documents contain bar charts?",
-                    "Summarize the environmental report",
-                    "What are the key findings?"
-                ]
-            )
-        
-        with gr.TabItem("📤 Upload"):
-            gr.Markdown("Upload new documents to the knowledge base")
-            file_uploader = gr.File(
-                label="Upload Documents",
-                file_count="multiple",
-                file_types=["image", ".pdf", ".txt", ".md"]
-            )
-            upload_btn = gr.Button("Process Documents", variant="primary")
-            status = gr.Markdown("Ready to upload.")
-            
-            upload_btn.click(fn=gradio_upload, inputs=[file_uploader], outputs=[status])
-
 # -------------------------------
 # Initialize and Run
 # -------------------------------
 if __name__ == "__main__":
-    print("🚀 Starting Multimodal RAG System...")
+    print("🚀 Starting Multimodal RAG API...")
     
     # Build initial database if data folder exists
     folder = "data"
@@ -565,19 +503,11 @@ if __name__ == "__main__":
         print(f"\n📂 Found '{folder}' folder, building database...")
         build_or_update_qdrant_store(folder)
     
-    # Launch both Flask and Gradio
-    from werkzeug.serving import run_simple
-    from werkzeug.middleware.dispatcher import DispatcherMiddleware
-    
-    # Mount Gradio at root, Flask API at /api
-    application = DispatcherMiddleware(
-        gradio_ui.launch(prevent_thread_lock=True, show_error=True),
-        {'/api': flask_app}
-    )
-    
-    print("\n✅ Server starting on http://0.0.0.0:7860")
-    print("   - Gradio UI: http://0.0.0.0:7860")
-    print("   - Flask API: http://0.0.0.0:7860/api/query")
-    print("   - Health Check: http://0.0.0.0:7860/api/health")
+    print("\n✅ Flask API starting on http://0.0.0.0:7860")
+    print("   Endpoints:")
+    print("   - GET  / (Home/Docs)")
+    print("   - POST /query")
+    print("   - POST /upload")
+    print("   - GET  /health\n")
     
-    run_simple('0.0.0.0', 7860, application, use_reloader=False, use_debugger=True)
+    app.run(host='0.0.0.0', port=7860, debug=False)