Update app.py

app.py

@@ -173,6 +173,26 @@ doctor_dashboard_interface = gr.Interface(
     outputs="text",
 )
 
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+
+# Load the doctor consultant model
+doctor_consultant = pipeline("text-generation", model="ahmed-7124/dgptAW")
+
+# New function for Doctor Consultant (Text Generation for Consultation)
+def doctor_consultation(query):
+    response = doctor_consultant(query, max_length=200, num_return_sequences=1)
+    return response[0]['generated_text']
+
+# Adding the doctor consultant interface to Gradio
+doctor_consultation_interface = gr.Interface(
+    fn=doctor_consultation,
+    inputs=gr.Textbox(label="Ask the Doctor (Consultation Query)", placeholder="Enter your medical query here..."),
+    outputs="text",
+    title="Doctor Consultant",
+    description="Get expert advice from the Doctor Consultant powered by AI. Ask any medical-related questions and get responses."
+)
+
 # Gradio App Layout
 with gr.Blocks() as app:
     gr.Markdown("# Medico GPT")
@@ -200,5 +220,8 @@ with gr.Blocks() as app:
     
     with gr.Tab("Doctor Dashboard"):
         doctor_dashboard_interface.render()
+    
+    with gr.Tab("Doctor Consultant"):
+        doctor_consultation_interface.render()
 
 app.launch(share=True)