import gradio as gr
from conversation import make_conversation
import random
import time

global USERNAME
global PASSWORD
global INPUT
global OUTPUT
global SOURCE
global DOCS

# def auth_function(username, password):
#     USERNAME = username
#     user_name = username
#     return username == password


# def make_conversation(message, history):
#     INPUT = message
#     text_, source, docs = run(message)
#     OUTPUT = text_
#     SOURCE = source
#     DOCS = docs

#     # print("INPUT: ", INPUT)
#     # print("OUTPUT: ", OUTPUT)
#     # print("SOURCE: ", SOURCE)
#     # print("DOCS: ", DOCS)

#     for i in range(len(text_)):
#         time.sleep(0.001)
#         yield text_[: i+1]

# We are making some changes, the model will not work at the moment.

with gr.Blocks(css="style.css") as demo:
    gr.Markdown("""
   
    
    # Dr. V AI
    
    Dr. V.AI is an experimental large language model tailored for Ophthalmology, integrating an extensive knowledge base.

    ## Potential Use Cases
    
    - **Real-time Assistant:** Functions as an assistant during patient consultations, providing instant diagnosis and treatment plans by analyzing patent information and historical records.
    - **Hospital Triage Support:** Aids in the triaging process within a hospital setting, streamlining patient flow based on urgency and severity.
    - **Research Integration:** Cites and incorporates recent research findings, enhancing diagnostic accuracy and treatment recommendations.
    - **Conversational Interface:** Facilitates seamless communication with organizational patient data, ensuring efficient and secure information exchange.
    - **Normal Patient Detection:** Identifies normal cases more effectively, minimizing unnecessary specialist intervention and optimizing resource allocation.

    ## Feedback
    
    - We value your feedback to enhance and refine the model. Please share your thoughts on the usability, accuracy, and any suggestions for improvement.

    ## Next Steps
    
    - Continual refinement based on user feedback and ongoing developments in Ophthalmology.
    - Including Image modality like Fundus.

    Thank you for engaging with Dr. V.AI. Your insights are instrumental in advancing the capabilities of this experimental Ophthalmology Language Model.
    """)
    gr.ChatInterface(make_conversation).queue()

demo.launch()