Spaces:
Runtime error
Runtime error
import gradio as gr | |
import sqlite3 | |
import pandas as pd | |
import time | |
import huggingface_hub | |
import shutil | |
import os | |
import datetime | |
from apscheduler.schedulers.background import BackgroundScheduler | |
from rag_output import rag_response | |
TOKEN = os.environ.get('HFW_TOKEN') | |
def llm_response(message, history): | |
res = rag_response(message) | |
for i in range(len(res)): | |
time.sleep(0.02) | |
yield res[: i+1] | |
# return res | |
def vote(response: gr.LikeData): | |
if response.liked: | |
add_review(1, response.value) | |
else: | |
add_review(0, response.value) | |
examples = ["What are the recommended NPK dosage for maize varieties?", | |
# "What are the recommended chemical treatments to control army worms in wheat crops?", | |
"Heavy rains are predicted next week. Is my rice crop ready for this, or should I harvest early?", | |
"What crops can I grow during the dry season to use water more efficiently?", | |
"How can I improve the health of my soil after a wheat harvest, using natural methods?", | |
# "Are there crop rotation techniques that can reduce fertilizer needs for barley?" | |
] | |
# js_func = """ | |
# function refresh() { | |
# const url = new URL(window.location); | |
# if (url.searchParams.get('__theme') !== 'light') { | |
# url.searchParams.set('__theme', 'light'); | |
# window.location.href = url.href; | |
# } | |
# } | |
# """ | |
description = "Hi! I am akṣara, an AI agronomist and I am here to help you with agriculture advisories for crops like paddy, wheat, maize, Sorghum, Barley, Cotton, Sugarcane, Soybean and Millets for the Indian subcontinent." | |
title = "akṣara" | |
theme = gr.themes.Soft(primary_hue="sky",) | |
chatbot = gr.Chatbot(height="450px", | |
show_copy_button=True, | |
show_label=False, | |
avatar_images=("user.webp","cropin.png")) | |
textbox = gr.Textbox(placeholder="Ask akṣara...", | |
min_width=300) | |
with gr.Blocks(theme=theme, title=title, css="footer {visibility: hidden}") as akshara: | |
gr.HTML("""<h1 style='font-family: sans-serif; text-align: center; font-size: 34px'> | |
<i style='color: #04A5D9' >akṣara</i> </h1>""") | |
gr.HTML("""<h3 style='font-family: sans-serif; text-align: left'> | |
Welcome! </h3>""") | |
# with gr.Column(): | |
chatbot.like(vote, None, None) | |
gr.ChatInterface(fn=llm_response, | |
examples=examples, | |
cache_examples=False, | |
chatbot=chatbot, | |
description=description, | |
retry_btn="Retry", | |
undo_btn="Undo", | |
clear_btn="Clear", | |
submit_btn="Ask", | |
textbox=textbox | |
) | |
gr.HTML("""<h3 style='font-family: sans-serif; text-align: left'> | |
Disclaimer: Beta Test version #1.0 - akṣara is still in the beta testing stage and please verify information with agronomy experts or local extensions officers | |
""") | |
def display_ui(): | |
akshara.launch() | |
if __name__ == "__main__": | |
display_ui() | |
pass | |