Initial commit: Quantum-API with FastAPI and Streamlit integration

api/endpoints/codelama.py

@@ -0,0 +1,16 @@
+from fastapi import FastAPI
+import subprocess  # Import subprocess module to run system commands
+import ollama
+
+app = FastAPI()
+
+@app.get("/")
+def read_root():
+    return {"message": "Welcome to the Quantum-API/codelama"}
+
+# Add other endpoints as needed for interacting with ollama
+@app.get("/run-codelama")
+def run_codelama():
+    # You can call ollama commands here or use subprocess
+    result = subprocess.run(["ollama", "run", "codellama:latest"], capture_output=True, text=True)
+    return {"result": result.stdout}

api/endpoints/ollama.py

@@ -0,0 +1,29 @@
+from fastapi import FastAPI, APIRouter
+from pydantic import BaseModel
+import ollama  # Make sure Ollama is installed and available
+
+app = FastAPI()
+
+# Define request model for the input data
+class UserInput(BaseModel):
+    question: str
+
+# Function to generate responses using Ollama
+def get_ollama_response(user_input: str) -> str:
+    try:
+        # Run Ollama model for generating a response
+        response = ollama.chat(model="llama", messages=[{"role": "user", "content": user_input}])
+        return response['text']  # Ensure you're extracting the response text from Ollama's response
+    except Exception as e:
+        return f"Error processing request: {str(e)}"
+
+# Create an API router
+router = APIRouter()
+
+@router.post("/ollama-response")
+async def ollama_response(user_input: UserInput):
+    response = get_ollama_response(user_input.question)
+    return {"response": response}
+
+# Include router into the FastAPI app
+app.include_router(router)

app/app.py

@@ -0,0 +1,33 @@
+import streamlit as st
+import requests
+
+# URL of the FastAPI backend endpoint
+API_URL_OLAMA = "http://localhost:7860/ollama-response"
+API_URL_CODELAMA = "http://localhost:7860/run-codelama"
+
+def main():
+    st.title("Quantum-API Chat Interface with Olama and CodeLlama")
+
+    user_input = st.text_input("Ask a question:")
+
+    if user_input:
+        if st.button("Chat with Olama"):
+            # Make a POST request to the FastAPI server for Olama
+            response = requests.post(API_URL_OLAMA, json={"question": user_input})
+            if response.status_code == 200:
+                # Display the response from Olama
+                st.write(f"Olama says: {response.json()['response']}")
+            else:
+                st.error("Error contacting Olama API.")
+
+        if st.button("Run Code with CodeLlama"):
+            # Make a GET request to the FastAPI server for CodeLlama
+            response = requests.get(API_URL_CODELAMA)
+            if response.status_code == 200:
+                # Display the response from CodeLlama
+                st.write(f"CodeLlama result: {response.json()['result']}")
+            else:
+                st.error("Error contacting CodeLlama API.")
+        
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,7 @@
+fastapi
+uvicorn
+streamlit
+olama
+codelama
+pennyLane
+qiskit