KisanAIAdvisor / app.py
MHamzaShahid's picture
Update app.py
929bb9f verified
import gradio as gr
import os
import requests
API_KEY = os.getenv("GROQ_API_KEY", "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")
def ask_kisanai(question):
url = "https://api.groq.com/openai/v1/chat/completions"
headers = {
"Authorization": f"Bearer {API_KEY}",
"Content-Type": "application/json"
}
system_prompt = "You are KisanAIAdvisor for Pakistani farmers. Give practical agriculture advice in simple language."
data = {
"model": "llama-3.1-8b-instant",
"messages": [
{"role": "system", "content": system_prompt},
{"role": "user", "content": question}
],
"temperature": 0.3,
"max_tokens": 300
}
try:
response = requests.post(url, headers=headers, json=data)
if response.status_code == 200:
return response.json()['choices'][0]['message']['content']
else:
return f"API Error: {response.status_code}"
except Exception as e:
return f"Error: {str(e)}"
# Simple interface
demo = gr.Interface(
fn=ask_kisanai,
inputs=gr.Textbox(label="Ask about agriculture", placeholder="Type in Urdu or English..."),
outputs=gr.Textbox(label="Answer"),
title="🌾 KisanAIAdvisor",
examples=[
["What crops grow best in Punjab?"],
["Meri fasal mein keera lag gaya hai"],
["How to test soil quality?"]
]
)
if __name__ == "__main__":
demo.launch()