new changes made

app.py

@@ -29,10 +29,14 @@ app.add_middleware(
 
 @app.post("/upload")
 async def upload_files(
-    session_id: str = Form(default_factory=lambda: str(uuid.uuid4())),
+    session_id: str = Form(default=""),  # Allow empty string
     files: List[UploadFile] = File(...)
 ):
-    """Upload and process files"""
+    """Upload and process files with automatic session management"""
+    # Generate session ID if not provided
+    if not session_id.strip():
+        session_id = str(uuid.uuid4())
+    
     if not files:
         raise HTTPException(400, detail="No files uploaded")
     
@@ -42,11 +46,16 @@ async def upload_files(
             content = await f.read()
             pairs.append((f.filename or "unnamed", content))
         except Exception as e:
-            logger.error(f"Failed to read {f.filename}: {e}")
+            logger.error(f"Failed to process {f.filename}: {str(e)}")
             continue
     
     added = ingestion.ingest_files(session_id, pairs)
-    return {"status": "success", "documents_added": added, "session_id": session_id}
+    return {
+        "status": "success",
+        "session_id": session_id,
+        "documents_added": added,
+        "message": f"Use this session_id for queries: {session_id}"
+    }
 
 @app.post("/query")
 async def query(

pipelines.py

@@ -3,15 +3,16 @@ os.environ["HAYSTACK_TELEMETRY_ENABLED"] = "False"
 os.environ["HF_HOME"] = "/app/cache"
 os.environ["HUGGINGFACE_HUB_CACHE"] = "/app/cache"
 import logging
-from haystack.utils import Secret
 import time
+from haystack.utils import Secret
 from haystack.dataclasses import Document
 from haystack.document_stores.in_memory import InMemoryDocumentStore
 from haystack.components.embedders import SentenceTransformersDocumentEmbedder, SentenceTransformersTextEmbedder
 from haystack.components.retrievers.in_memory import InMemoryEmbeddingRetriever
 from haystack.components.rankers import SentenceTransformersSimilarityRanker
-from haystack_integrations.components.generators.google_ai import GoogleAIGeminiGenerator
 from haystack.components.preprocessors import DocumentSplitter
+import google.generativeai as genai
+from google.generativeai import types
 
 # Set up logging
 logger = logging.getLogger(__name__)
@@ -27,7 +28,7 @@ doc_embedder = SentenceTransformersDocumentEmbedder(
 text_embedder = SentenceTransformersTextEmbedder(
     model="BAAI/bge-base-en-v1.5"
 )
-retriever = InMemoryEmbeddingRetriever(document_store=document_store, top_k=3)  # Reduce top_k
+retriever = InMemoryEmbeddingRetriever(document_store=document_store, top_k=3) 
 
 def initialize_ranker():
     attempts = 0
@@ -59,16 +60,30 @@ except Exception as e:
             return {"documents": documents[:3]}
     reranker = DummyRanker()
 
-# Initialize generator
-generator = GoogleAIGeminiGenerator(
-    api_key=Secret.from_env_var("GOOGLE_API_KEY"),
-    model="gemini-1.5-flash-latest",
-    safety_settings={
-        "HARASSMENT": "BLOCK_NONE",
-        "HATE_SPEECH": "BLOCK_NONE",
-        "SEXUAL": "BLOCK_NONE",
-        "DANGEROUS": "BLOCK_NONE"
-    }
+# Configure GenAI SDK
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+
+# Initialize generator with safety settings
+generator = genai.GenerativeModel(
+    model_name="gemini-2.0-flash",
+    safety_settings=[
+        types.SafetySetting(
+            category=types.HarmCategory.HARM_CATEGORY_HARASSMENT,
+            threshold=types.HarmBlockThreshold.BLOCK_NONE
+        ),
+        types.SafetySetting(
+            category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+            threshold=types.HarmBlockThreshold.BLOCK_NONE
+        ),
+        types.SafetySetting(
+            category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+            threshold=types.HarmBlockThreshold.BLOCK_NONE
+        ),
+        types.SafetySetting(
+            category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+            threshold=types.HarmBlockThreshold.BLOCK_NONE
+        )
+    ]
 )
 
 splitter = DocumentSplitter(
@@ -167,8 +182,8 @@ def query_rag(question: str, session_id: str):
         prompt = f"Context:\n{context}\n\nQuestion: {question}\nAnswer:"
         
         # Handle generator response safely
-        response = generator.run(parts=[prompt])
-        answer = response.get("replies", [""])[0] if response and response.get("replies") else "No response generated"
+        response = generator.generate_content(prompt)
+        answer = response.text if response and hasattr(response, 'text') else "No response generated"
         
         # Format sources
         sources = [

requirements.txt

@@ -11,3 +11,5 @@ python-dotenv
 haystack-ai
 google-ai-haystack
 sentence-transformers
+
+google-generativeai