Upload app.py

app.py

@@ -0,0 +1,62 @@
+# -*- coding: utf-8 -*-
+"""Untitled25.ipynb
+
+Automatically generated by Colaboratory.
+
+Original file is located at
+    https://colab.research.google.com/drive/1i4BW958q6NUrUaJ2bAgNM8-N1pLSeP-Q
+"""
+
+pip install openai streamlit
+
+
+
+import openai
+import streamlit as st
+
+# Initialize OpenAI API and server base
+openai.api_key = "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"
+openai.api_base = "http://zanino.millennium.berkeley.edu:8000/v1"
+
+# Set Streamlit layout
+st.set_page_config(layout="wide")
+
+# Function to get response from Gorilla Server
+def get_gorilla_response(prompt, model):
+    try:
+        completion = openai.ChatCompletion.create(
+            model=model,
+            messages=[{"role": "user", "content": prompt}]
+        )
+        return completion.choices[0].message.content
+    except Exception as e:
+        print("An error occurred:", e)
+
+def main():
+    st.title("Gorilla LLM")
+    input_prompt = st.text_area("Enter your prompt:")
+
+    model_options = ('gorilla-7b-hf-v1', 'gorilla-mpt-7b-hf-v0')
+    option = st.selectbox('Select a model:', model_options)
+
+    if st.button("Create"):
+        if len(input_prompt) > 0:
+            result = get_gorilla_response(prompt=input_prompt, model=option)
+            st.write(result)
+            # ...
+
+if st.button("Gorilla Magic"):
+    if len(input_prompt) > 0:
+        # ...
+        with col2:
+            code_result = extract_code_from_output(result)
+            if option == "gorilla-7b-hf-v1":
+                st.subheader("Generated Output")
+                st.code(code_result, language='python')
+            elif option == "gorilla-mpt-7b-hf-v0":
+                lines = code_result.split('\\n')
+                for line in lines[:-1]:
+                    st.code(line, language='python')
+            # ...
+            run_generated_code(file_path)
+