Upload 2 files

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import requests
 import cohere
-
+from translate import Translator
 
 def get_weather_data():
     lat = 27.0238
@@ -16,34 +16,35 @@ def get_weather_data():
 def generate_prompt(data):
     weather_json = get_weather_data()
     prompt = \
-    f'''Category: {data["Category"]}
-Crop Type: {data["Crop"]}
-Query Type: {data["QueryType"]}
-State: {data["StateName"].lower()}
-District: {data["DistrictName"].lower()}
+    f'''State: Rajasthan
 Max Temprature: {weather_json["temp_max"]}
 Min Temprature: {weather_json["temp_min"]}
 Humidity: {weather_json["humidity"]}
 Weather Description: {weather_json["description"]}
-Question: {data["QueryTranslated"]}
+Context: Through increased use of soil testing and plant analyses, micronutrient deficiencies have been verified in many soils. Some reasons limiting the incidental additions of micronutrients include.High-yield crop demands remove micronutrients from the soil. Increased use of high-analysis NPK fertilizers containing lower quantities of micronutrient contaminants. Advances in fertilizer technology reduce the residual addition of micronutrients.
+Question: {data}
 '''
     return prompt
 
 def get_response(prompt):
     co = cohere.Client('EoYqxEa60C0EEeKadblGW8NE94geVCEE75lDqySe')
-    response = co.generate(  
+    new_prompt = generate_prompt(prompt)
+    response = co.generate(
         model='command-xlarge-nightly',  
-        prompt = prompt,  
-        max_tokens=40,  
-        temperature=0.6,  
-        stop_sequences=["--"]
+        prompt = new_prompt,  
+        max_tokens = 1000,
+        temperature = 0.6,  
+        stop_sequences = ["--"]
     )
-    print(response[0])
-    return response.generations[0].text
+    translator = Translator(to_lang="hi")
+    translation = translator.translate(response.generations[0].text)
+    return translation
 
-outputs = gr.outputs.Textbox()
-iface = gr.Interface(fn=get_response, 
-                     inputs=[gr.Textbox(label = 'First Code snippet')], 
-                     outputs=[gr.Textbox(label = 'Summary of first Code snippet')],
-                     description='The similarity score')
-iface.launch()
+title = """<h1 align="center">🌱 Farmer Queries LLM | किसान प्रश्न LLM 🌾</h1>"""
+with gr.Blocks(css="""#col_container {margin-left: auto; margin-right: auto;} #chatbot {height: 520px; overflow: auto;}""") as demo:
+    gr.HTML(title)
+    input1 = gr.Textbox(label = 'प्रश्न पूछो')
+    output1 = gr.Textbox(label = 'उत्तर')
+    btn = gr.Button("जमा करे")
+    btn.click(get_response, [input1], output1)
+demo.launch()

requirements.txt

@@ -1 +1,2 @@
-cohere
+cohere
+translate