Initial commit

backend/copywriter.py

@@ -0,0 +1,24 @@
+# backend/copywriter.py
+from .vector_store import VectorStore
+from .brand_style import BrandStyle
+import openai
+
+class Copywriter:
+    def __init__(self):
+        self.vector_store = VectorStore()
+        self.brand_style = BrandStyle()
+    
+    def generate_copy(self, request):
+        # Move the generation logic from main.py here
+        similar = self.vector_store.search(request.prompt, request.content_type)
+        
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {"role": "system", "content": self.brand_style.get_prompt(request)},
+                {"role": "user", "content": f"Prompt: {request.prompt}\n\nSimilar examples:\n{similar}"}
+            ],
+            temperature=0.7
+        )
+        
+        return response.choices[0].message.content