File size: 1,234 Bytes
f6e5cd9
5a6a740
 
 
9db0dbd
5a6a740
 
 
 
a1e8dca
 
3b3a846
a1e8dca
 
 
5a6a740
 
 
 
a1e8dca
 
5a6a740
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
a1e8dca
5a6a740
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import os
import gradio as gr
import requests

ollama_url = '0.0.0.0'

console_output = []

def call_ollama_api(prompt):
    try:
        response = requests.post(
            f'{ollama_url}/generate',
            json={"prompt": prompt}
        )
        response.raise_for_status()  # Raise an exception for HTTP errors
        result = response.json()
        output = result.get('text', '')
        console_output.append(f"Prompt: {prompt}\nResponse: {output}")
        return output
    except requests.exceptions.RequestException as e:
        error_message = f"Error: {e}"
        console_output.append(error_message)
        return error_message

def chat(prompt):
    output = call_ollama_api(prompt)
    return output

def show_console_output():
    return "\n\n".join(console_output)

iface = gr.Interface(
    fn=chat,
    inputs="text",
    outputs="text",
    title="Ollama Chat",
    description="Chat with Ollama API and see the responses."
)

console_iface = gr.Interface(
    fn=show_console_output,
    inputs=None,
    outputs="text",
    title="Console Output",
    description="See the console outputs here."
)

# Launch a tabbed interface
gr.TabbedInterface([iface, console_iface], ["Chat", "Console"]).launch()