Update app.py

app.py

@@ -4,6 +4,7 @@ from huggingface_hub import list_models
 import os
 from dotenv import load_dotenv
 import requests
+import json
 
 load_dotenv(verbose=True)
 groqkey = os.environ.get("GROQ_API_KEY")
@@ -46,13 +47,13 @@ def list_private_models(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthTo
     # if the user is not logged in, oauth_token will be None
     gr.Textbox(oauth_token)
     if oauth_token is None:
-        return "Please log in to list private models.", gr.update(visible=False), gr.update(visible=False)
+        return "Please log in to list private models.", gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
     #models = [
         #f"{model.id} ({'private' if model.private else 'public'})"
         #for model in list_models(author=profile.username, token=oauth_token.token)
     #]
     #return "Models:\n\n" + "\n - ".join(models) + ".", gr.update(visible=True)
-    return profile.username, gr.update(visible=True), gr.update(visible=True)
+    return profile.username, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 
 def process_eprag(prompt):
@@ -65,6 +66,23 @@ def process_eprag(prompt):
         return rtn
 
 
+def process_agent(prompt):
+    if prompt == "":
+        return "プロンプトを入力してください。", "プロンプトは必須です。"
+    else:
+        url = 'https://www.ryhintl.com/crewai/autogen?qry='+prompt
+        res = requests.get(url)
+        rtn = res.content.decode('utf-8')
+        
+        parsed_data = json.loads(rtn)
+        content_list = [entry['content'] for entry in parsed_data['chat_history']]
+
+        for content in content_list:
+            mycontent = content
+           
+        return mycontent
+
+
 with gr.Blocks() as agentic:
     gr.Markdown(
         "# Gradio OAuth Space for Agentic RAG"
@@ -105,7 +123,25 @@ with gr.Blocks() as agentic:
             inputs=[eprag_input],
             outputs=[eprag_output]
         )
+
+    with gr.Tab("AGENT", visible=False) as tab_agentic:
+        
+        
+        gr.Markdown("# 🗞️ AGENTIC AUTOGEN")
+        
+        with gr.Row():
+            agent_input = gr.Textbox(label="プロンプト", type="text")
+            
+        with gr.Row():
+            agent_output = gr.Textbox(label="Agentアシスタントの応答")
+            
+        submit_button = gr.Button("AGENTICプロセス", variant="primary")
+        submit_button.click(
+            process_agent,
+            inputs=[agent_input],
+            outputs=[agent_output]
+        )
                 
-    agentic.load(list_private_models, inputs=None, outputs=[m2, tab_llm, tab_eprag])
+    agentic.load(list_private_models, inputs=None, outputs=[m2, tab_llm, tab_eprag, tab_agentic])
 
 agentic.launch()