Update app.py

app.py

@@ -1,29 +1,37 @@
 import gradio as gr
-import requests
-from huggingface_hub import InferenceClient
 import os
+import requests
+from langchain_groq import ChatGroq
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.output_parsers import StrOutputParser
 
 def preprocess_city_name_and_generate_insight(city_name):
-    token = os.getenv("HF_TOKEN")
-    if not token:
-        return city_name, "Hugging Face token is not set in the environment variables."
-    
-    try:
-
-        client = InferenceClient(token=token)
+    groq_api_key = os.getenv("GROQ_API_KEY")
+    if not groq_api_key:
+        return city_name, "GROQ API key is not set in the environment variables."
 
-        prompt = f"Generate a short, interesting fact about {city_name} in less than 10 words."
+    try:
+        chat = ChatGroq(api_key=groq_api_key, model_name="llama3-70b-8192")
 
-        result = client(inputs=prompt, model_id="gpt2")
+        system_message = f"Generate a short, interesting fact about {city_name} in less than 10 words."
+        human_message = "{text}"
+        prompt = ChatPromptTemplate.from_messages(
+            [
+                ("system", system_message), ("human", human_message)
+            ]
+        )
+        chain = prompt | chat | StrOutputParser()
+        result = chain.invoke({"text": city_name})
 
-        insight = result['generated_text'].strip() if 'generated_text' in result else "No insight generated."
+        insight = result.strip() if result else "No insight generated."
     except Exception as e:
         print(f"Error generating insight: {str(e)}")
         insight = "Could not generate insight due to an error."
-    
+
     return city_name, insight
 
 
+
 def get_weather(city_name):
     corrected_city_name, city_insight = preprocess_city_name_and_generate_insight(city_name)
     API_Key = os.getenv("OPENWEATHER_API_KEY")