akurapati1/codeassistant

1import requests
2import json
3import gradio as gr
4
5url="http://localhost:11434/api/generate"
6
7headers={
8
9    'Content-Type':'application/json'
10}
11
12history=[]
13
14def generate_response(prompt):
15    history.append(prompt)
16    final_prompt="\n".join(history)
17
18    data={
19        "model":"codeguru",
20        "prompt":final_prompt,
21        "stream":False
22    }
23
24    response=requests.post(url,headers=headers,data=json.dumps(data))
25
26    if response.status_code==200:
27        response=response.text
28        data=json.loads(response)
29        actual_response=data['response']
30        return actual_response
31    else:
32        print("error:",response.text)
33
34
35interface=gr.Interface(
36    fn=generate_response,
37    inputs=gr.Textbox(lines=4,placeholder="Enter your Prompt"),
38    outputs="text"
39)
40interface.launch()
41