remove additional providers

app.py

@@ -6,8 +6,10 @@ from uuid import uuid4
 from gradio_client import Client, handle_file
 from utils.whisp_api import handle_geojson_upload
 from utils.retriever import retrieve_paragraphs
+from utils.generator import generate
 
 # Sample questions for examples 
+
 SAMPLE_QUESTIONS = {
     "Deforestation Analysis": [
         "What are the main deforestation hotspots in Ecuador?",
@@ -26,6 +28,7 @@ SAMPLE_QUESTIONS = {
     ]
 }
 
+# Initialize Chat
 
 def start_chat(query, history):
     """Start a new chat interaction"""
@@ -67,8 +70,9 @@ async def chat_response(query, history, method, country, uploaded_file):
     else: 
         try: 
             response = f"Based on EUDR reports for {country}, I can help you understand deforestation patterns and compliance requirements. Your question: '{query}' is being analyzed..."
-            retrieved_info = retrieve_paragraphs(query)[0]
-            response = retrieved_info
+            retrieved_paragraphs = retrieve_paragraphs(query)
+            response = generate(query=query, context=retrieve_paragraphs)
+
         except Exception as e:
             response = f"Error retrieving information: {str(e)}"
 

utils/generator.py

@@ -6,9 +6,6 @@ from typing import List, Dict, Any, Union
 from dotenv import load_dotenv
 
 # LangChain imports
-from langchain_openai import ChatOpenAI
-from langchain_anthropic import ChatAnthropic
-from langchain_cohere import ChatCohere
 from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
 from langchain_core.messages import SystemMessage, HumanMessage
 
@@ -37,10 +34,10 @@ def getconfig(configfile_path: str):
 def get_auth(provider: str) -> dict:
     """Get authentication configuration for different providers"""
     auth_configs = {
-        "openai": {"api_key": os.getenv("OPENAI_API_KEY")},
+        #"openai": {"api_key": os.getenv("OPENAI_API_KEY")},
         "huggingface": {"api_key": os.getenv("HF_TOKEN")},
-        "anthropic": {"api_key": os.getenv("ANTHROPIC_API_KEY")},
-        "cohere": {"api_key": os.getenv("COHERE_API_KEY")},
+        #"anthropic": {"api_key": os.getenv("ANTHROPIC_API_KEY")},
+        #"cohere": {"api_key": os.getenv("COHERE_API_KEY")},
     }
     
     if provider not in auth_configs:
@@ -76,25 +73,25 @@ def get_chat_model():
     }
     logging.info(f"provider is {PROVIDER}")
     
-    if PROVIDER == "openai":
-        return ChatOpenAI(
-            model=MODEL,
-            openai_api_key=auth_config["api_key"],
-            **common_params
-        )
-    elif PROVIDER == "anthropic":
-        return ChatAnthropic(
-            model=MODEL,
-            anthropic_api_key=auth_config["api_key"],
-            **common_params
-        )
-    elif PROVIDER == "cohere":
-        return ChatCohere(
-            model=MODEL,
-            cohere_api_key=auth_config["api_key"],
-            **common_params
-        )
-    elif PROVIDER == "huggingface":
+    # if PROVIDER == "openai":
+    #     return ChatOpenAI(
+    #         model=MODEL,
+    #         openai_api_key=auth_config["api_key"],
+    #         **common_params
+    #     )
+    # elif PROVIDER == "anthropic":
+    #     return ChatAnthropic(
+    #         model=MODEL,
+    #         anthropic_api_key=auth_config["api_key"],
+    #         **common_params
+    #     )
+    # elif PROVIDER == "cohere":
+    #     return ChatCohere(
+    #         model=MODEL,
+    #         cohere_api_key=auth_config["api_key"],
+    #         **common_params
+    #    )
+    if PROVIDER == "huggingface":
         # Initialize HuggingFaceEndpoint with explicit parameters
         llm = HuggingFaceEndpoint(
             repo_id=MODEL,